#!/bin/bash

# - Author: jiruyi
# - Filename: run.sh
# - Date: Tue Oct  9 09:16:18 CST 2018

# - Description:
#   first fine tune the new-added layers only

set -e
WORK_PATH=$(dirname $(readlink -f $0))
TOOLS=$CAFFE_ROOT/build/tools
LOG=$WORK_PATH/logs/log-server-`date +%Y-%m-%d-%H-%M-%S`-device_$1.log

#stage_first
$TOOLS/caffe train \
	-solver "$WORK_PATH/prototxts/stage_first/solver.prototxt"  \
    -weights "$WORK_PATH/model-zoo/Inits/init_tree.caffemodel" \
    -log_dir "$WORK_PATH/logs/" \
    -gpu $1 \
    2>&1 | tee $LOG

